package com.xy.lock;

import org.springframework.data.redis.core.DefaultTypedTuple;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * @author xxy
 * @date 2022/1/5 13:15
 */
@Component
public class RedisLock {

    private final String lockKey = "lock";

    private final StringRedisTemplate stringRedisTemplate;

    public RedisLock(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }


    /**
     * 尝试获取锁
     *
     * @param reqId      请求id
     * @param expireTime 过期时间
     * @return
     */
    public boolean tryGetLock(String reqId, long expireTime) {

        //这里利用redis的set命令
        //redis（判断是否存在，添加key，设置过期时间）
        // 判断是否存在 key 为 lockKey 的 key , 不存在 设置 key 为 lockKey 的 key 并设置 请求 id 和 过期时间 单位为 秒
        while (true) {
            if (stringRedisTemplate.boundValueOps(lockKey).setIfAbsent(reqId, expireTime, TimeUnit.SECONDS)) {
                return true;
            }
        }

    }


    /**
     * 释放锁
     *
     * @param reqId 请求 id
     * @return
     */
    public boolean unLock(String reqId) {

        // 这里使用Lua脚本保证原子性操作
        String script = "if  redis.call('get', KEYS[1]) == ARGV[1] then " +
                "return redis.call('del', KEYS[1]) " +
                "else return 0 end";
        DefaultRedisScript<Long> defaultRedisScript = new DefaultRedisScript<>(script, Long.class);
        return new Long(1)
                .equals(stringRedisTemplate
                        .execute(defaultRedisScript, Collections.singletonList(lockKey), reqId));

    }


    /**
     * 创建续命子线程
     * @param time 操作预期耗时
     * @param reqId 唯一标识
     * @return
     */
    public Thread createLifeThread(long time, String reqId) {

        return new Thread(() -> {

            while (true) {
                try {
                    TimeUnit.SECONDS.sleep(time * 2 / 3);
                    //重置时间
                    String script = "if redis.call('get', KEYS[1]) == ARGV[1] then " +
                            "return redis.call('expire', KEYS[1],ARGV[2]) " +
                            "else return '0' end";

                    List args = new ArrayList();
                    args.add(reqId);
                    args.add(time);


                    DefaultRedisScript<Long> defaultRedisScript =
                            new DefaultRedisScript<>(script, Long.class);
                    stringRedisTemplate.execute(defaultRedisScript,
                            Collections.singletonList(lockKey), args);

                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

        });

    }


}
